pytorch mistakes